package com.mnnu;

import java.util.Iterator;
import java.util.function.Consumer;

/**
 * 单向链表
 */
public class SinglyLinkedList implements Iterable<Integer> { //链表整体
    private Node head = new Node(666, null); //头指针 (带哨兵)

    /**
     * 迭代器
     * @return 迭代器对象
     */
    @Override
    public Iterator<Integer> iterator() {
        return new Iterator<Integer>() {
            Node p = head.next;
            @Override
            public boolean hasNext() { //判断是否有下一个元素
                return p != null;
            }

            @Override
            public Integer next() { //返回当前值，并指向下一个元素
                int v = p.value;
                p = p.next;
                return v;
            }
        };
    }

    /**
     * 节点类
     */
    private class Node {
        int value; //值
        Node next; //下一个节点指针

        public Node(int value, Node next) {
            this.value = value;
            this.next = next;
        }
    }

    /**
     * 向链表头部添加
     * @param value  待添加的值
     */
    public void addFirst(int value) {
        Insert(0, value);
    }

    /**
     * 遍历链表
     */
    public void loop1(Consumer<Integer> consumer) {
        Node p = head.next;
        while(p != null) {
            consumer.accept(p.value);
            p = p.next;
        }
    }
    public void loop2(Consumer<Integer> consumer) {
        for(Node p = head.next; p != null; p = p.next) {
            consumer.accept(p.value);
        }
    }

    /**
     * 找到链表最后一个节点
     * @return 最后节点
     */
    private Node findLast() {
        Node p;
        for(p = head; p.next != null; p = p.next) {

        }
        return p;
    }

    /**
     * 向链表尾部添加
     * @param value 待添加的值
     */
    public void addLast(int value) {
        Node last = findLast();
        last.next = new Node(value, null);
    }

    /**
     * 根据索引查找节点
     * @param index 索引
     * @return 节点
     */
    private Node findNode(int index) {
        int i = -1;
        for(Node p = head; p != null; p = p.next, i++) {
            if(i == index) {
                return p;
            }
        }
        return null;
    }

    /**
     * 根据索引查找节点值
     * @param index 索引
     * @return 节点值
     */
    public int get(int index) {
        Node node = findNode(index);
        if(node == null) {
            throw IllegalIndex(index);
        }
        return node.value;
    }

    private IllegalArgumentException IllegalIndex(int index) {
        return new IllegalArgumentException
                (String.format("index [%d] 不存在", index));
    }

    /**
     * 根据索引位置插入节点
     * @param index 索引
     * @param value 节点值
     */
    public void Insert(int index, int value) {
        Node prev = findNode(index - 1); //找上一节点
        if(prev == null) {
            throw IllegalIndex(index);
        }
        prev.next = new Node(value, prev.next);
    }

    /**
     * 移除第一个节点
     */
    public void removeFirst() {
        remove(0);
    }

    /**
     * 移除结点
     * @param index
     */
    public void remove(int index) {
        Node prev = findNode(index - 1); //上一个节点
        Node remove = findNode(index); //将删除的节点
        if(remove == null) {
            throw IllegalIndex(index);
        }
        prev.next = remove.next;
    }
}

